class Config(object):
    batch_size = 128
    embedding_dim = 128
    hidden_dim = 256
    learning_rate = 1e-3
    max_poetry_length = 260  # length of poetry
    model_param_path = 'param/param.pth'
    train_loss_path = 'param/loss.pth'
    epoch = 1
